import datetime
import os.path
import sys

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager


class Driver:
    driver = ""
    def __init__(self):
        options = webdriver.ChromeOptions()
        self.driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()),options=options)
        self.driver.implicitly_wait(5)
    def getScreeShot(self): #创建屏幕截图
        dirname = datetime.datetime.now().strftime("%Y-%m-%d")
        if not os.path.exists("../images/"+dirname): #判断dirname文件夹是否已经存在，若不存在则创建文件夹
            os.mkdir("../images/"+dirname) #给每一天都生成一个文件夹路径，文件夹名就用当天日期，当天获取到的截图就都放到对应文件夹里

        #获取调用方法名：sys._getframe().f_back.f_code.co_name
        filename = sys._getframe().f_back.f_code.co_name+"-"+datetime.datetime.now().strftime("%Y-%m-%d-%H%M%S")+".png"
        #最终的截图文件名就类似于：LoginSucTest-2025-06-11-141000.png
        self.driver.save_screenshot("../images/"+dirname+"/"+filename)

BlogDriver = Driver()